.detach().cpu().numpy()[0]